﻿// content-script.js

// 检测页面中的所有图片并在每张图片右下角添加下载提示
function addDownloadIcons() {
	const images = document.querySelectorAll('img');
	images.forEach(img => {
		// 创建下载图标
		const downloadIcon = document.createElement('img');
		downloadIcon.src = chrome.runtime.getURL('img/download-icon.png');  // 使用插件中的图标
		downloadIcon.style.position = 'absolute';
		downloadIcon.style.bottom = '5px';
		downloadIcon.style.right = '5px';
		downloadIcon.style.width = '20px';
		downloadIcon.style.height = '20px';
		downloadIcon.style.cursor = 'pointer';
		downloadIcon.title = '下载图片';

		// 创建一个容器来放置图标
		const container = document.createElement('div');
		container.style.position = 'relative';
		container.style.display = 'inline-block';

		// 将图片和图标放入容器中
		container.appendChild(img.cloneNode());
		container.appendChild(downloadIcon);

		// 替换原始图片
		img.parentNode.replaceChild(container, img);

		// 监听下载图标点击事件
		downloadIcon.addEventListener('click', () => {
			chrome.runtime.sendMessage({ action: "downloadImage", imageUrl: img.src });
		});
	});
}

// 监听点击事件，触发图片检测
document.addEventListener('click', function(event) {
	if (event.target.id === 'addDownloadIconsButton') {
		addDownloadIcons();
	}
});
